STL unordered_set的使用

• C++
• unordered_set
• STL

template < class Key, 
           class Hash = hash<Key>,        
           class Pred = equal_to<Key>,    
           class Alloc = allocator<Key>   
           > class unordered_set;

1.使用前先定义

(1)定义元素类

class Entry
{
private:
    string key;
    Value value;
public:
    Entry(const string str,Value * value){
        key = str;
        this->value.a = value->a;
        this->value.b = value->b;
    }
    string getK()
    {
        return key;
    }
    Value getV()
    {
        return value;
    }
};

(2)定义比较函数

struct NodeEquality
{
    inline bool operator()(Entry* const& _s1, Entry* const& _s2) const
    {
        return _s1->getV().a == _s2->getV().a && _s1->getV().b == _s2->getV().b;
    }
};

(3)定义HASH函数

namespace std
{
    template<> struct hash<Entry*>
    {
        typedef Entry* argument_type;
        typedef std::size_t result_type;
        result_type operator()(argument_type const& s) const
        {
            return hash<string>()(s->getK());
        }
    };
}

(4)定义unordered_set

unordered_set<Entry *,std::hash<Entry *>, NodeEquality> myset;

2.插入新元素

myset.insert(new Entry("123",new Value(2,4)));

说明

插入时首先会根据hash函数取hash，若有重复，则通过比较函数判断内容是否相同，若仍然相同，则不会插入，否则插入该项(由于key相同，新项会链表形式挂在上一项的后面)

3.查找

auto it = myset.find(new Entry("123",new Value(2,8)));

说明

查找时会同时根据hash函数以及比较函数判断，两者都相同时判断找到此项。

4.举个例子

#include <iostream>
#include <unordered_set>

using namespace std;

class Value
{
public:
    int a;
    int b;
    Value(int a, int b)
    {
        this->a = a;
        this->b = b;
    }
    Value()
    {
        a = b = 0;
    }
};

class Entry
{
private:
    string key;
    Value value;
public:
    Entry(const string str,Value * value){
        key = str;
        this->value.a = value->a;
        this->value.b = value->b;
    }
    string getK()
    {
        return key;
    }
    Value getV()
    {
        return value;
    }
};

struct NodeEquality
{
    inline bool operator()(Entry* const& _s1, Entry* const& _s2) const
    {
        return _s1->getV().a == _s2->getV().a && _s1->getV().b == _s2->getV().b;
    }
};

namespace std
{
    template<> struct hash<Entry*>
    {
        typedef Entry* argument_type;
        typedef std::size_t result_type;
        result_type operator()(argument_type const& s) const
        {
            return hash<string>()(s->getK());
        }
    };
}

int main(int argc, char *argv[])
{
    unordered_set<Entry *,std::hash<Entry *>, NodeEquality> myset;
    myset.insert(new Entry("123",new Value(2,4)));
    myset.insert(new Entry("123",new Value(2,5)));
    myset.insert(new Entry("1235",new Value(6,7)));
    myset.insert(new Entry("1232",new Value(2,8)));

    for(auto it = myset.begin(); it != myset.end(); it++)
    {   
        cout << "Key:" <<(*it)->getK() << " Value:(" <<(*it)->getV().a << "," <<(*it)->getV().b << ")" << endl;
    }

    auto it = myset.find(new Entry("123",new Value(2,4)));
    if(it != myset.end() )
        cout << (*it)->getV().a << " " <<(*it)->getV().b << endl;
    else
        cout << "NULL" << endl;

    system("pause");
    return 0;
}

赏

嘘寒问暖不如来笔巨款

支付宝

微信

本文总阅读量 次

本文由 Yu Zhang 发表于 Yu Zhang's Blog ,采用署名-非商业性使用-禁止演绎 3.0进行许可。
非商业转载请注明作者及出处。商业转载请联系作者本人。